# coding: utf-8
import matplotlib.pylab as plt
import numpy as np


def step_function(x):
    return np.array(x > 0, dtype=np.int)  # 阶跃函数的结果只有0或1


X = np.arange(-5.0, 5.0, 0.1)  # arange表示安排，查看API学学该函数
Y = step_function(X)
plt.plot(X, Y)  # 使用x和y进行绘图
plt.ylim(-0.1, 1.1)  # 指定图中绘制的y轴的范围
plt.show()
